from basic_model_train import main_data as basic_model_train
from config import config
from lora_data_set_pre_group import main_data as lora_data_set_pre_group
from lora_model_train import train_lora_main
from gen_voc import main_data as gen_voc_func


def multi_lora():
    for one_lora in range(0, config["lora_level"]):

        if one_lora == 0:
            # 训练基础模型
            basic_model_train()

        lora_data_set_pre_group(one_lora)

        train_lora_main(one_lora)


if __name__ == '__main__':
    # gen_voc_func()
    multi_lora()
